package com.moyoutian.leetcode;

/**
 * 110. 平衡二叉树
 * <p>
 * 给定一个二叉树，判断它是否是高度平衡的二叉树。
 * <p>
 * 本题中，一棵高度平衡二叉树定义为：
 * <p>
 * 一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。
 * <p>
 * <p>
 * <p>
 * 示例 1：
 * <p>
 * 输入：root = [3,9,20,null,null,15,7]
 * 输出：true
 * <p>
 * 示例 2：
 * <p>
 * 输入：root = [1,2,2,3,3,null,null,4,4]
 * 输出：false
 * <p>
 * 示例 3：
 * <p>
 * 输入：root = []
 * 输出：true
 * <p>
 * <p>
 * <p>
 * 提示：
 * <p>
 * 树中的节点数在范围 [0, 5000] 内
 * -104 <= Node.val <= 104
 */
public class Demo110 {

    public static class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode() {
        }

        TreeNode(int val) {
            this.val = val;
        }

        TreeNode(int val, TreeNode left, TreeNode right) {
            this.val = val;
            this.left = left;
            this.right = right;
        }
    }


    public static void main(String[] args) {
        Integer[] root = {3, 9, 20, null, null, 15, 7};
        TreeNode treeNode = createAVLTree(root, 0, 0);
        System.out.println(isBalanced(treeNode));
    }

    public static boolean isBalanced(TreeNode root) {
        return heBalanced(root) == -1;
    }

    public static int heBalanced(TreeNode root) {
        //树的左子树和右子树的高度差的绝对值不能超过1.
        //如果递归遍历。递归序的话第一次访问左树肯定不知道右树的高度。所以递归应该向上返回结果在根据跟结点统计。所以应该是后续遍历。
        if (root == null) {
            return 0;
        }
        int leftHeight = heBalanced(root.left);
        int rightHeight = heBalanced(root.right);
        if (leftHeight == -1 || rightHeight == -1 || Math.abs(leftHeight - rightHeight) > 1) {
            return -1;
        } else {
            return Math.max(leftHeight, rightHeight) + 1;
        }
    }


    /**
     * @param integers 数组
     * @param in       下标
     * @param h        深度
     */
    public static TreeNode createAVLTree(Integer[] integers, int in, int h) {
        if (in < integers.length && integers[in] != null) {
            TreeNode root = new TreeNode();
            root.left = createAVLTree(integers, in + 1 + h * 2, h + 1);
            root.right = createAVLTree(integers, in + 2 + h * 2, h + 1);
            return root;
        }
        return null;
    }

}
